#pragma once
#include <iostream>
#include <string>
#include <unistd.h>
#include <wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/Util.hpp"
#include "../comm/Logger.hpp"
// 这个类是只负责编译，不关心程序执行结果正确与否
namespace ns_compile
{
    using namespace ns_util;
    using namespace ns_log;
    class Compiler
    {
    private:
        static bool IsExeExit(const std::string &filename)
        {
            struct stat statbuff;
            return stat(PathUtil::Exe(filename).c_str(), &statbuff) == 0;
        }
    public:
        // test.cpp
        // ./temp/test.cpp
        // ./temp/test.exe
        // ./temp/test.stderr
        // 暂时，没有规划，日志信息
        static bool Compile(const std::string &filename)
        {
            // 检查源文件是否存在
            if (!FileUtil::FileIsExit(PathUtil::Src(filename)))
            {
                LOG(LogLevel::ERROR) << "源文件: " << PathUtil::Src(filename).c_str() << " 不存在";
                return false;
            }

            pid_t id = fork();
            if (id < 0)
            {
                LOG(LogLevel::ERROR) << "子进程创建失败";
                exit(1); // 暂时
            }
            else if (id == 0)
            {
                // 0. 将权限掩码设置为0
                umask(0);
                // 1. 将stderr 重定向
                int stderrfd = open(PathUtil::compile_error(filename).c_str(), O_CREAT | O_WRONLY, 0644);
                if (stderrfd < 0)
                {
                    LOG(LogLevel::ERROR) << "打开stderr文件失败";
                    exit(2);
                }
                // 3. 重定向
                dup2(stderrfd, STDERR_FILENO);
                close(stderrfd); // 重定向后立即关闭原始文件描述符！
                // 4. 程序替换
                // 子进程 g++ -o filename.exe filename.src -std=c++11
                execlp("/usr/bin/g++", "g++", "-o", PathUtil::Exe(filename).c_str(), PathUtil::Src(filename).c_str(),\
                       "-std=c++11", "-DCOMPILER_ONLINE", nullptr); 
                LOG(LogLevel::ERROR) << "exec error";
                exit(3); // 成功不会走到这
            }
            waitpid(id, nullptr, 0);
            if (!IsExeExit(filename))
            {
                LOG(LogLevel::ERROR) << "编译失败";
                return false;
            }
            LOG(LogLevel::INFO) << "编译: " << PathUtil::Src(filename).c_str() << " -> "
                                << PathUtil::Exe(filename).c_str() << " 成功";
            return true;
        }
    };
}